package xyz.magicjson.core.adapter;

import lombok.extern.slf4j.Slf4j;
import xyz.magicjson.core.exception.ESBException;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import javax.annotation.PreDestroy;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.ServiceLoader;
import java.util.concurrent.ConcurrentHashMap;

/**
 * 协议适配器管理器，负责管理所有协议适配器。
 * 通过SPI机制和Spring自动注入发现并注册协议适配器。
 */
@Slf4j
@Component
public class ProtocolAdapterManager {

    /**
     * 存储所有协议适配器的映射
     */
    private final Map<String, ProtocolAdapter> adapterMap = new ConcurrentHashMap<>();
    
    /**
     * 自动注入所有协议适配器
     */
    @Autowired
    private List<ProtocolAdapter> adapters;
    
    /**
     * 初始化协议适配器管理器，发现并注册所有协议适配器
     */
    @PostConstruct
    public void init() {
        log.info("初始化协议适配器管理器");
        
        // 注册Spring注入的适配器
        registerSpringAdapters();
        
        // 通过SPI机制发现适配器
        discoverAdaptersBySpi();
        
        // 启动所有适配器
        startAllAdapters();
        
        log.info("协议适配器管理器初始化完成，已注册 {} 个适配器", adapterMap.size());
    }
    
    /**
     * 注册Spring注入的适配器
     */
    private void registerSpringAdapters() {
        if (adapters != null) {
            for (ProtocolAdapter adapter : adapters) {
                String protocol = adapter.getProtocol();
                log.info("注册Spring适配器: {}", protocol);
                adapterMap.put(protocol, adapter);
            }
        }
    }
    
    /**
     * 通过SPI机制发现适配器
     */
    private void discoverAdaptersBySpi() {
        ServiceLoader<ProtocolAdapter> serviceLoader = ServiceLoader.load(ProtocolAdapter.class);
        for (ProtocolAdapter adapter : serviceLoader) {
            String protocol = adapter.getProtocol();
            if (!adapterMap.containsKey(protocol)) {
                log.info("通过SPI发现适配器: {}", protocol);
                adapterMap.put(protocol, adapter);
            } else {
                log.warn("适配器已存在，跳过SPI发现的适配器: {}", protocol);
            }
        }
    }
    
    /**
     * 启动所有适配器
     */
    private void startAllAdapters() {
        for (Map.Entry<String, ProtocolAdapter> entry : adapterMap.entrySet()) {
            String protocol = entry.getKey();
            ProtocolAdapter adapter = entry.getValue();
            
            // 初始化适配器
            Map<String, Object> config = new HashMap<>();
            config.put("enabled", true);
            adapter.initialize(config);
            
            // 启动适配器
            adapter.start();
            log.info("适配器启动成功: {}", protocol);
        }
    }
    
    /**
     * 释放资源
     */
    @PreDestroy
    public void destroy() {
        log.info("关闭协议适配器管理器");
        
        // 停止所有适配器
        for (Map.Entry<String, ProtocolAdapter> entry : adapterMap.entrySet()) {
            String protocol = entry.getKey();
            ProtocolAdapter adapter = entry.getValue();
            
            try {
                adapter.stop();
                log.info("适配器关闭成功: {}", protocol);
            } catch (Exception e) {
                log.error("适配器关闭失败: {}, 异常: {}", protocol, e.getMessage(), e);
            }
        }
        
        adapterMap.clear();
    }
    
    /**
     * 获取指定协议的适配器
     *
     * @param protocol 协议名称
     * @return 协议适配器
     * @throws ESBException 如果找不到指定协议的适配器
     */
    public ProtocolAdapter getAdapter(String protocol) {
        ProtocolAdapter adapter = adapterMap.get(protocol);
        if (adapter == null) {
            throw new ESBException(String.format("找不到协议 [%s] 的适配器", protocol));
        }
        return adapter;
    }
    
    /**
     * 注册协议适配器
     *
     * @param adapter 协议适配器
     */
    public void registerAdapter(ProtocolAdapter adapter) {
        String protocol = adapter.getProtocol();
        log.info("注册协议适配器: {}", protocol);
        
        adapterMap.put(protocol, adapter);
        
        // 初始化适配器
        Map<String, Object> config = new HashMap<>();
        config.put("enabled", true);
        adapter.initialize(config);
        
        // 启动适配器
        adapter.start();
    }
    
    /**
     * 注销协议适配器
     *
     * @param protocol 协议名称
     */
    public void unregisterAdapter(String protocol) {
        ProtocolAdapter adapter = adapterMap.remove(protocol);
        if (adapter != null) {
            log.info("注销协议适配器: {}", protocol);
            adapter.stop();
        }
    }
    
    /**
     * 获取所有已注册的协议
     *
     * @return 协议名称列表
     */
    public Map<String, ProtocolAdapter> getAllAdapters() {
        return new HashMap<>(adapterMap);
    }
} 